# 导包
import requests
import config


# 创建接口类
class StaffAPI:
    # 初始化
    def __init__(self):
        self.url = config.BASE_URL + "/api/sys/user"

    # 员工添加
    # 参数数据不变的，直接写在当前代码文件即可
    # 参数数据是变化，我们需要定义参数接受从外部传入的数据
    def insert_staff(self, insert_data, token):
        return requests.post(url=self.url, json=insert_data, headers={"Authorization": token})

    def select_staff(self, staff_id, token):
        return requests.get(url=self.url + "/" + staff_id, headers={"Authorization": token})

    def update_staff(self, staff_id, token, update_data):
        return requests.put(url=self.url + "/" + staff_id, json=update_data, headers={"Authorization": token})

    def delete_staff(self, staff_id, token):
        return requests.delete(url=self.url + "/" + staff_id, headers={"Authorization": token})


if __name__ == '__main__':
    # 注意：
    # （1）添加员工时，手机号必须唯一
    # （2）当前测试代码是使用手动关联方式解决前置条件登录成功（ 先运行登录测试代码、登录成功之后、手动复制data中的数据即可）
    insert_data = {
        "username": "jack886622",
        "mobile": "15906012001",
        "workNumber": "jack15906012001"
    }
    response = StaffAPI().insert_staff(insert_data=insert_data, token='af8f31bf-58c3-462a-b360-3b66d0ec775e')
    print(response.json())
